package com.github.niefy.modules.oss.cloud;
import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.github.niefy.common.exception.RRException;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
/**
 * 阿里云存储
 *
 * @author Mark sunlightcs@gmail.com
 */
public class AliyunAbstractCloudStorageService extends AbstractCloudStorageService {
  private OSS client;
  public AliyunAbstractCloudStorageService(CloudStorageConfig config) {
    this.config = config;
    //初始化
    init();
  }
  private void init() {
    client = new OSSClientBuilder().build(config.getAliyunEndPoint(), config.getAliyunAccessKeyId(), config.getAliyunAccessKeySecret());
  }
  @Override
  public String upload(byte[] data, String path) {
    return upload(new ByteArrayInputStream(data), path);
  }
  @Override
  public String upload(InputStream inputStream, String path) {
    try {
      client.putObject(config.getAliyunBucketName(), path, inputStream);
    }
    catch (Exception e) {
      throw new RRException("上传文件失败，请检查配置信息", e);
    }
    return config.getAliyunDomain() + "/" + path;
  }
  @Override
  public String uploadSuffix(byte[] data, String suffix) {
    return upload(data, getPath(config.getAliyunPrefix(), suffix));
  }
  @Override
  public String uploadSuffix(InputStream inputStream, String suffix) {
    return upload(inputStream, getPath(config.getAliyunPrefix(), suffix));
  }
}
